how to split data into 3 sets (train validation and test)